import os
import uuid
import shutil
class Myfiles():
    # 指定文件目录
    def __init__(self,upload_dir='../static/upload/'):
        self.up_dir=upload_dir

    # 创建文件夹
    def mk_dir(self,userid):
        # 判断是否存在
        path = self.up_dir+str(userid)+'/'
        if not os.path.exists(path):
            # 创建文件
            os.mkdir(path)
            
    # 文件重命名
    def rename(self,file):
        filename = uuid.uuid4().hex + os.path.splitext(file)[1]
        return filename

    # 获取指定文件目录下的所有文件
    def show_all(self,userid):
        files = os.listdir(self.up_dir+str(userid)+'/')
        return files

    # 删除单个目录
    def remove_file(self,userid,file):
        path = self.up_dir+str(userid)+'/'+file
        os.remove(path)

    # 删除整个目录
    def remove_dir(self,userid):
        path = self.up_dir+str(userid)+'/'
        # 判断文件是否存在
        if os.path.exists(path):
            # 存在删除文件夹及所有的文件
            shutil.rmtree(path)



myf = Myfiles()
# myf.mk_dir(5)
# myf.rename('1561.jpg')
# file = myf.show_all(1)
# print(file)
# myf.remove_file(1,'ef1cbcd5686640a2aec0e28da91b0fcf.png')
# myf.remove_dir(5)
